package com.sky.service.impl;

import com.sky.dto.GoodsSalesDTO;
import com.sky.entity.Orders;
import com.sky.mapper.OrderMapper;
import com.sky.mapper.UserMapper;
import com.sky.service.ReportService;
import com.sky.service.WorkspaceService;
import com.sky.vo.*;
import org.apache.commons.lang.StringUtils;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 报表业务层实体类
 *
 * @author liuqi
 * @Date 2023/11/11
 */
@Service
public class ReportServiceImpl implements ReportService {

    @Autowired
    private OrderMapper orderMapper;
    @Autowired
    private UserMapper userMapper;
    @Autowired
    private WorkspaceService workspaceService;

    /**
     * 统计营业额
     */
    @Override
    public TurnoverReportVO turnoverStatistics(LocalDate begin, LocalDate end) {
        // 条件查询订单
        // 条件：状态为已完成、每天的营业额
        List<LocalDate> dateList = getLocalDateList(begin, end);
        // 营业额集合
        List<Double> amountList = new ArrayList<>();
        // 遍历日期集合，查询每天的营业额
        for (LocalDate localDate : dateList) {
            // 00:00:00
            LocalDateTime beginTime = LocalDateTime.of(localDate, LocalTime.MIN);
            // 23:59:59
            LocalDateTime endTime = LocalDateTime.of(localDate, LocalTime.MAX);
            // 封装查询条件到map集合中
            Map params = new HashMap();
            // 状态
            params.put("status", Orders.COMPLETED);
            // 开始时间
            params.put("begin", beginTime);
            // 结束时间
            params.put("end", endTime);
            // 调用mapper层查询营业额
            Double amount = this.orderMapper.sumByMap(params);
            // 将每天的营业额存入集合
            amountList.add(amount == null ? 0 : amount);
        }
        // 拼接返回值
        TurnoverReportVO turnoverReportVO = new TurnoverReportVO();
        // 日期
        turnoverReportVO.setDateList(StringUtils.join(dateList, ","));
        // 营业额
        turnoverReportVO.setTurnoverList(StringUtils.join(amountList, ","));
        // 返回结果
        return turnoverReportVO;
    }


    /**
     * 统计用户数量
     *
     * @param begin
     * @param end
     */
    @Override
    public UserReportVO getUserStatistics(LocalDate begin, LocalDate end) {
        // 根据时间区间，获取时间集合
        List<LocalDate> dateList = getLocalDateList(begin, end);
        // 用户总量集合
        List<Integer> totalUserList = new ArrayList<>();
        // 新增用户集合
        List<Integer> newUserList = new ArrayList<>();
        // 遍历时间集合
        for (LocalDate localDate : dateList) {
            // 00:00:00
            LocalDateTime beginTime = LocalDateTime.of(localDate, LocalTime.MIN);
            // 23:59:59
            LocalDateTime endTime = LocalDateTime.of(localDate, LocalTime.MAX);
            // 新增用户数量
            Integer newUser = getUserCount(beginTime, endTime);
            newUserList.add(newUser);
            // 总用户数量
            Integer totalUser = getUserCount(null, endTime);
            totalUserList.add(totalUser);
        }
        // 拼接返回值
        UserReportVO userReportVO = new UserReportVO();
        // 时间
        userReportVO.setDateList(StringUtils.join(dateList, ","));
        // 总用户数量
        userReportVO.setTotalUserList(StringUtils.join(totalUserList, ","));
        // 新增用户数量
        userReportVO.setNewUserList(StringUtils.join(newUserList, ","));

        return userReportVO;
    }

    /**
     * 订单数据统计
     */
    @Override
    public OrderReportVO getOrderStatistics(LocalDate begin, LocalDate end) {
        // 获取时间集合
        List<LocalDate> dateList = getLocalDateList(begin, end);

        // 总订单集合
        List<Integer> orderCountList = new ArrayList<>();
        // 有效订单集合
        List<Integer> validOrderCountList = new ArrayList<>();
        // 遍历时间集合
        for (LocalDate localDate : dateList) {
            // 00:00:00
            LocalDateTime beginTime = LocalDateTime.of(localDate, LocalTime.MIN);
            // 23:59:59
            LocalDateTime endTime = LocalDateTime.of(localDate, LocalTime.MAX);

            // 有效订单集合
            Integer validCount = getOrdersCountByTime(beginTime, endTime, Orders.COMPLETED);
            validOrderCountList.add(validCount);
            // 总订单集合
            Integer orderCount = getOrdersCountByTime(beginTime, endTime, null);
            orderCountList.add(orderCount);
        }
        // 获取总订单数量
        Integer totalOrderCount = orderCountList.stream().reduce(Integer::sum).get();
        // 获取有效订单数
        Integer validOrderCount = validOrderCountList.stream().reduce(Integer::sum).get();
        // 获取订单完成率
        Double orderCompletionRate = 0.0;
        if (totalOrderCount != 0) {
            orderCompletionRate = validOrderCount.doubleValue() / totalOrderCount;
        }
        // 封装最后结果
        OrderReportVO orderReportVO = new OrderReportVO();
        // 时间集合
        orderReportVO.setDateList(StringUtils.join(dateList, ","));
        // 总订单集合
        orderReportVO.setOrderCountList(StringUtils.join(orderCountList, ","));
        // 有效订单集合
        orderReportVO.setValidOrderCountList(StringUtils.join(validOrderCountList, ","));
        // 总订单数量
        orderReportVO.setTotalOrderCount(totalOrderCount);
        // 有效订单数量
        orderReportVO.setValidOrderCount(validOrderCount);
        // 订单完成率
        orderReportVO.setOrderCompletionRate(orderCompletionRate);
        // 返回最后结果
        return orderReportVO;
    }

    /**
     * 统计商品销量前10名
     */
    @Override
    public SalesTop10ReportVO getSalesTop10(LocalDate begin, LocalDate end) {
        // 开始时间
        LocalDateTime beginTime = LocalDateTime.of(begin, LocalTime.MIN);
        // 结束时间
        LocalDateTime endTime = LocalDateTime.of(end, LocalTime.MAX);
        // 查询订单中商品销量前十的菜品或者套餐
        List<GoodsSalesDTO> goodsSalesDTOList = this.orderMapper.getSalesTop10(beginTime,endTime);
        // 菜品或套餐名称集合
        List<String> nameList = goodsSalesDTOList.stream().map(a -> a.getName()).collect(Collectors.toList());
        // 销量集合
        List<Integer> numberList = goodsSalesDTOList.stream().map(a -> a.getNumber()).collect(Collectors.toList());
        // 封装结果
        SalesTop10ReportVO salesTop10ReportVO = new SalesTop10ReportVO();
        // 名称
        salesTop10ReportVO.setNameList(StringUtils.join(nameList,","));
        // 销量
        salesTop10ReportVO.setNumberList(StringUtils.join(numberList,","));
        // 返回最终结果
        return salesTop10ReportVO;
    }

    /**
     *  导出近30天的运营数据报表
     */
    @Override
    public void exportBusinessData(HttpServletResponse response) {
        // 获取30天前的时间
        // 30天前的开始时间
        LocalDate beginDate = LocalDate.now().minusDays(30);
        // 开始时间的最小时间
        LocalDateTime beginDateMinTime = LocalDateTime.of(beginDate, LocalTime.MIN);
        // 1天前的结束时间
        LocalDate endDate = LocalDate.now().minusDays(1);
        // 结束时间的最大时间
        LocalDateTime endDateMinTime = LocalDateTime.of(endDate, LocalTime.MAX);
        // 获取数据
        BusinessDataVO businessData = this.workspaceService.getBusinessData(beginDateMinTime, endDateMinTime);
        // 加载模版
        InputStream resourceAsStream = this.getClass().getClassLoader().getResourceAsStream("template/运营数据报表模板.xlsx");
        try {
            // 创建工作簿
            XSSFWorkbook workbook = new XSSFWorkbook(resourceAsStream);
            // 拿到sheet页
            XSSFSheet sheet = workbook.getSheetAt(0);
            // 赋值日期
            sheet.getRow(1).getCell(1).setCellValue(beginDate + " 至 " + endDate);
            // 获取第四行
            XSSFRow row4 = sheet.getRow(3);
            // 营业额
            row4.getCell(2).setCellValue(businessData.getTurnover().toString());
            // 订单完成率
            row4.getCell(4).setCellValue(businessData.getOrderCompletionRate());
            // 新增用户数
            row4.getCell(6).setCellValue(businessData.getNewUsers().toString());
            // 获取第五行
            XSSFRow row5 = sheet.getRow(4);
            // 有效订单
            row5.getCell(2).setCellValue(businessData.getValidOrderCount().toString());
            // 平均客单价
            row5.getCell(4).setCellValue(businessData.getUnitPrice());

            // 循环30次,设置30天的数据
            for (int i = 0; i < 30; i++) {
                // 获取日期
                beginDate = beginDate.plusDays(1);
                // 获取当前的最大时间和最小时间
                // 开始时间 == 最小时间
                LocalDateTime minTime = LocalDateTime.of(beginDate, LocalTime.MIN);
                // 结束时间 == 最大时间
                LocalDateTime maxTime = LocalDateTime.of(beginDate, LocalTime.MAX);
                // 调用接口查询数据
                BusinessDataVO data = this.workspaceService.getBusinessData(minTime, maxTime);
                // 获取每行
                XSSFRow row = sheet.getRow(7 + i);
                // 当天日期
                row.getCell(1).setCellValue(beginDate.toString());
                // 当天营业额
                row.getCell(2).setCellValue(data.getTurnover().toString());
                // 当天有效订单
                row.getCell(3).setCellValue(data.getValidOrderCount().toString());
                // 当天订单完成率
                row.getCell(4).setCellValue(data.getOrderCompletionRate());
                // 当天平均客单价
                row.getCell(5).setCellValue(data.getUnitPrice());
                // 当天新增用户数
                row.getCell(6).setCellValue(data.getNewUsers().toString());
            }
            // 下载excel
            ServletOutputStream outputStream = response.getOutputStream();
            // 通过浏览器下载
            workbook.write(outputStream);
            // 刷新流
            outputStream.flush();
            // 关流
            outputStream.close();
            workbook.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private Integer getOrdersCountByTime(LocalDateTime beginTime, LocalDateTime endTime, Integer status) {
        Map params = new HashMap();
        // 订单状态
        params.put("status", status);
        // 开始时间
        params.put("begin", beginTime);
        // 结束时间
        params.put("end", endTime);
        // 获取有效订单数量
        return this.orderMapper.countByMap(params);
    }


    private Integer getUserCount(LocalDateTime beginTime, LocalDateTime endTime) {
        // 封装查询条件到map集合中
        Map params = new HashMap();
        // 开始时间
        params.put("begin", beginTime);
        // 结束时间
        params.put("end", endTime);
        // 调用service获取用户数量
        return this.userMapper.countByMap(params);
    }

    /**
     * 根据时间区间获取时间集合
     */
    private List<LocalDate> getLocalDateList(LocalDate begin, LocalDate end) {
        List<LocalDate> dateList = new ArrayList<>();
        dateList.add(begin);
        // 将开始日期到结束日期每天的日期存入集合中
        // 如果开始日期和结束日期一致，结束循环
        while (!begin.equals(end)) {
            // 开始时间+1
            begin = begin.plusDays(1);
            dateList.add(begin);
        }
        return dateList;
    }
}
